Printing a flattened list using list comprehension in Python